"""
方便随时进行测试
"""

import torch
import numpy as np

# device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

# print(device.type)
list2d = [
    [2,0],
    [4,0]
]

arr2d = np.array(list2d)

print(arr2d.shape)

print(np.mean(arr2d, axis=0))
